1 /* Copyright (c) 2007-2008 CSIRO 2 Copyright (c) 2007-2009 Xiph.Org Foundation 3 Copyright (c) 2008 Gregory Maxwell 4 Written by Jean-Marc Valin and Gregory Maxwell */ 5 /** 6 @file celt.h 7 @brief Contains all the functions for encoding and decoding audio 8 */ 9 10 /* 11 Redistribution and use in source and binary forms, with or without 12 modification, are permitted provided that the following conditions 13 are met: 14 15 - Redistributions of source code must retain the above copyright 16 notice, this list of conditions and the following disclaimer. 17 18 - Redistributions in binary form must reproduce the above copyright 19 notice, this list of conditions and the following disclaimer in the 20 documentation and/or other materials provided with the distribution. 21 22 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 23 ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 24 LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 25 A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER 26 OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 27 EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 28 PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 29 PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 30 LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 31 NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 32 SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 33 */ 34 35 #ifndef CELT_H 36 #define CELT_H 37 38 #include "opus_types.h" 39 #include "opus_defines.h" 40 #include "opus_custom.h" 41 #include "entenc.h" 42 #include "entdec.h" 43 #include "arch.h" 44 45 #ifdef __cplusplus 46 extern "C" { 47 #endif 48 49 #define CELTEncoder OpusCustomEncoder 50 #define CELTDecoder OpusCustomDecoder 51 #define CELTMode OpusCustomMode 52 53 #define LEAK_BANDS 19 54 55 typedef struct { 56 int valid; 57 float tonality; 58 float tonality_slope; 59 float noisiness; 60 float activity; 61 float music_prob; 62 float music_prob_min; 63 float music_prob_max; 64 int bandwidth; 65 float activity_probability; 66 float max_pitch_ratio; 67 /* Store as Q6 char to save space. */ 68 unsigned char leak_boost[LEAK_BANDS]; 69 } AnalysisInfo; 70 71 typedef struct { 72 int signalType; 73 int offset; 74 } SILKInfo; 75 76 #define __celt_check_mode_ptr_ptr(ptr) ((ptr) + ((ptr) - (const CELTMode**)(ptr))) 77 78 #define __celt_check_analysis_ptr(ptr) ((ptr) + ((ptr) - (const AnalysisInfo*)(ptr))) 79 80 #define __celt_check_silkinfo_ptr(ptr) ((ptr) + ((ptr) - (const SILKInfo*)(ptr))) 81 82 /* Encoder/decoder Requests */ 83 84 85 #define CELT_SET_PREDICTION_REQUEST 10002 86 /** Controls the use of interframe prediction. 87 0=Independent frames 88 1=Short term interframe prediction allowed 89 2=Long term prediction allowed 90 */ 91 #define CELT_SET_PREDICTION(x) CELT_SET_PREDICTION_REQUEST, __opus_check_int(x) 92 93 #define CELT_SET_INPUT_CLIPPING_REQUEST 10004 94 #define CELT_SET_INPUT_CLIPPING(x) CELT_SET_INPUT_CLIPPING_REQUEST, __opus_check_int(x) 95 96 #define CELT_GET_AND_CLEAR_ERROR_REQUEST 10007 97 #define CELT_GET_AND_CLEAR_ERROR(x) CELT_GET_AND_CLEAR_ERROR_REQUEST, __opus_check_int_ptr(x) 98 99 #define CELT_SET_CHANNELS_REQUEST 10008 100 #define CELT_SET_CHANNELS(x) CELT_SET_CHANNELS_REQUEST, __opus_check_int(x) 101 102 103 /* Internal */ 104 #define CELT_SET_START_BAND_REQUEST 10010 105 #define CELT_SET_START_BAND(x) CELT_SET_START_BAND_REQUEST, __opus_check_int(x) 106 107 #define CELT_SET_END_BAND_REQUEST 10012 108 #define CELT_SET_END_BAND(x) CELT_SET_END_BAND_REQUEST, __opus_check_int(x) 109 110 #define CELT_GET_MODE_REQUEST 10015 111 /** Get the CELTMode used by an encoder or decoder */ 112 #define CELT_GET_MODE(x) CELT_GET_MODE_REQUEST, __celt_check_mode_ptr_ptr(x) 113 114 #define CELT_SET_SIGNALLING_REQUEST 10016 115 #define CELT_SET_SIGNALLING(x) CELT_SET_SIGNALLING_REQUEST, __opus_check_int(x) 116 117 #define CELT_SET_TONALITY_REQUEST 10018 118 #define CELT_SET_TONALITY(x) CELT_SET_TONALITY_REQUEST, __opus_check_int(x) 119 #define CELT_SET_TONALITY_SLOPE_REQUEST 10020 120 #define CELT_SET_TONALITY_SLOPE(x) CELT_SET_TONALITY_SLOPE_REQUEST, __opus_check_int(x) 121 122 #define CELT_SET_ANALYSIS_REQUEST 10022 123 #define CELT_SET_ANALYSIS(x) CELT_SET_ANALYSIS_REQUEST, __celt_check_analysis_ptr(x) 124 125 #define OPUS_SET_LFE_REQUEST 10024 126 #define OPUS_SET_LFE(x) OPUS_SET_LFE_REQUEST, __opus_check_int(x) 127 128 #define OPUS_SET_ENERGY_MASK_REQUEST 10026 129 #define OPUS_SET_ENERGY_MASK(x) OPUS_SET_ENERGY_MASK_REQUEST, __opus_check_val16_ptr(x) 130 131 #define CELT_SET_SILK_INFO_REQUEST 10028 132 #define CELT_SET_SILK_INFO(x) CELT_SET_SILK_INFO_REQUEST, __celt_check_silkinfo_ptr(x) 133 134 /* Encoder stuff */ 135 136 int celt_encoder_get_size(int channels); 137 138 int celt_encode_with_ec(OpusCustomEncoder * OPUS_RESTRICT st, const opus_val16 * pcm, int frame_size, unsigned char *compressed, int nbCompressedBytes, ec_enc *enc); 139 140 int celt_encoder_init(CELTEncoder *st, opus_int32 sampling_rate, int channels, 141 int arch); 142 143 144 145 /* Decoder stuff */ 146 147 int celt_decoder_get_size(int channels); 148 149 150 int celt_decoder_init(CELTDecoder *st, opus_int32 sampling_rate, int channels); 151 152 int celt_decode_with_ec(OpusCustomDecoder * OPUS_RESTRICT st, const unsigned char *data, 153 int len, opus_val16 * OPUS_RESTRICT pcm, int frame_size, ec_dec *dec, int accum); 154 155 #define celt_encoder_ctl opus_custom_encoder_ctl 156 #define celt_decoder_ctl opus_custom_decoder_ctl 157 158 159 #ifdef CUSTOM_MODES 160 #define OPUS_CUSTOM_NOSTATIC 161 #else 162 #define OPUS_CUSTOM_NOSTATIC static OPUS_INLINE 163 #endif 164 165 static const unsigned char trim_icdf[11] = {126, 124, 119, 109, 87, 41, 19, 9, 4, 2, 0}; 166 /* Probs: NONE: 21.875%, LIGHT: 6.25%, NORMAL: 65.625%, AGGRESSIVE: 6.25% */ 167 static const unsigned char spread_icdf[4] = {25, 23, 2, 0}; 168 169 static const unsigned char tapset_icdf[3]={2,1,0}; 170 171 #ifdef CUSTOM_MODES 172 static const unsigned char toOpusTable[20] = { 173 0xE0, 0xE8, 0xF0, 0xF8, 174 0xC0, 0xC8, 0xD0, 0xD8, 175 0xA0, 0xA8, 0xB0, 0xB8, 176 0x00, 0x00, 0x00, 0x00, 177 0x80, 0x88, 0x90, 0x98, 178 }; 179 180 static const unsigned char fromOpusTable[16] = { 181 0x80, 0x88, 0x90, 0x98, 182 0x40, 0x48, 0x50, 0x58, 183 0x20, 0x28, 0x30, 0x38, 184 0x00, 0x08, 0x10, 0x18 185 }; 186 toOpus(unsigned char c)187 static OPUS_INLINE int toOpus(unsigned char c) 188 { 189 int ret=0; 190 if (c<0xA0) 191 ret = toOpusTable[c>>3]; 192 if (ret == 0) 193 return -1; 194 else 195 return ret|(c&0x7); 196 } 197 fromOpus(unsigned char c)198 static OPUS_INLINE int fromOpus(unsigned char c) 199 { 200 if (c<0x80) 201 return -1; 202 else 203 return fromOpusTable[(c>>3)-16] | (c&0x7); 204 } 205 #endif /* CUSTOM_MODES */ 206 207 #define COMBFILTER_MAXPERIOD 1024 208 #define COMBFILTER_MINPERIOD 15 209 210 extern const signed char tf_select_table[4][8]; 211 212 #if defined(ENABLE_HARDENING) || defined(ENABLE_ASSERTIONS) 213 void validate_celt_decoder(CELTDecoder *st); 214 #define VALIDATE_CELT_DECODER(st) validate_celt_decoder(st) 215 #else 216 #define VALIDATE_CELT_DECODER(st) 217 #endif 218 219 int resampling_factor(opus_int32 rate); 220 221 void celt_preemphasis(const opus_val16 * OPUS_RESTRICT pcmp, celt_sig * OPUS_RESTRICT inp, 222 int N, int CC, int upsample, const opus_val16 *coef, celt_sig *mem, int clip); 223 224 void comb_filter(opus_val32 *y, opus_val32 *x, int T0, int T1, int N, 225 opus_val16 g0, opus_val16 g1, int tapset0, int tapset1, 226 const opus_val16 *window, int overlap, int arch); 227 228 #ifdef NON_STATIC_COMB_FILTER_CONST_C 229 void comb_filter_const_c(opus_val32 *y, opus_val32 *x, int T, int N, 230 opus_val16 g10, opus_val16 g11, opus_val16 g12); 231 #endif 232 233 #ifndef OVERRIDE_COMB_FILTER_CONST 234 # define comb_filter_const(y, x, T, N, g10, g11, g12, arch) \ 235 ((void)(arch),comb_filter_const_c(y, x, T, N, g10, g11, g12)) 236 #endif 237 238 void init_caps(const CELTMode *m,int *cap,int LM,int C); 239 240 #ifdef RESYNTH 241 void deemphasis(celt_sig *in[], opus_val16 *pcm, int N, int C, int downsample, const opus_val16 *coef, celt_sig *mem); 242 void celt_synthesis(const CELTMode *mode, celt_norm *X, celt_sig * out_syn[], 243 opus_val16 *oldBandE, int start, int effEnd, int C, int CC, int isTransient, 244 int LM, int downsample, int silence); 245 #endif 246 247 #ifdef __cplusplus 248 } 249 #endif 250 251 #endif /* CELT_H */ 252