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 vad_prob;
63 int bandwidth;
64 float activity_probability;
65 /* Store as Q6 char to save space. */
66 unsigned char leak_boost[LEAK_BANDS];
67 } AnalysisInfo;
68
69 typedef struct {
70 int signalType;
71 int offset;
72 } SILKInfo;
73
74 #define __celt_check_mode_ptr_ptr(ptr) ((ptr) + ((ptr) - (const CELTMode**)(ptr)))
75
76 #define __celt_check_analysis_ptr(ptr) ((ptr) + ((ptr) - (const AnalysisInfo*)(ptr)))
77
78 #define __celt_check_silkinfo_ptr(ptr) ((ptr) + ((ptr) - (const SILKInfo*)(ptr)))
79
80 /* Encoder/decoder Requests */
81
82
83 #define CELT_SET_PREDICTION_REQUEST 10002
84 /** Controls the use of interframe prediction.
85 0=Independent frames
86 1=Short term interframe prediction allowed
87 2=Long term prediction allowed
88 */
89 #define CELT_SET_PREDICTION(x) CELT_SET_PREDICTION_REQUEST, __opus_check_int(x)
90
91 #define CELT_SET_INPUT_CLIPPING_REQUEST 10004
92 #define CELT_SET_INPUT_CLIPPING(x) CELT_SET_INPUT_CLIPPING_REQUEST, __opus_check_int(x)
93
94 #define CELT_GET_AND_CLEAR_ERROR_REQUEST 10007
95 #define CELT_GET_AND_CLEAR_ERROR(x) CELT_GET_AND_CLEAR_ERROR_REQUEST, __opus_check_int_ptr(x)
96
97 #define CELT_SET_CHANNELS_REQUEST 10008
98 #define CELT_SET_CHANNELS(x) CELT_SET_CHANNELS_REQUEST, __opus_check_int(x)
99
100
101 /* Internal */
102 #define CELT_SET_START_BAND_REQUEST 10010
103 #define CELT_SET_START_BAND(x) CELT_SET_START_BAND_REQUEST, __opus_check_int(x)
104
105 #define CELT_SET_END_BAND_REQUEST 10012
106 #define CELT_SET_END_BAND(x) CELT_SET_END_BAND_REQUEST, __opus_check_int(x)
107
108 #define CELT_GET_MODE_REQUEST 10015
109 /** Get the CELTMode used by an encoder or decoder */
110 #define CELT_GET_MODE(x) CELT_GET_MODE_REQUEST, __celt_check_mode_ptr_ptr(x)
111
112 #define CELT_SET_SIGNALLING_REQUEST 10016
113 #define CELT_SET_SIGNALLING(x) CELT_SET_SIGNALLING_REQUEST, __opus_check_int(x)
114
115 #define CELT_SET_TONALITY_REQUEST 10018
116 #define CELT_SET_TONALITY(x) CELT_SET_TONALITY_REQUEST, __opus_check_int(x)
117 #define CELT_SET_TONALITY_SLOPE_REQUEST 10020
118 #define CELT_SET_TONALITY_SLOPE(x) CELT_SET_TONALITY_SLOPE_REQUEST, __opus_check_int(x)
119
120 #define CELT_SET_ANALYSIS_REQUEST 10022
121 #define CELT_SET_ANALYSIS(x) CELT_SET_ANALYSIS_REQUEST, __celt_check_analysis_ptr(x)
122
123 #define OPUS_SET_LFE_REQUEST 10024
124 #define OPUS_SET_LFE(x) OPUS_SET_LFE_REQUEST, __opus_check_int(x)
125
126 #define OPUS_SET_ENERGY_MASK_REQUEST 10026
127 #define OPUS_SET_ENERGY_MASK(x) OPUS_SET_ENERGY_MASK_REQUEST, __opus_check_val16_ptr(x)
128
129 #define CELT_SET_SILK_INFO_REQUEST 10028
130 #define CELT_SET_SILK_INFO(x) CELT_SET_SILK_INFO_REQUEST, __celt_check_silkinfo_ptr(x)
131
132 /* Encoder stuff */
133
134 int celt_encoder_get_size(int channels);
135
136 int celt_encode_with_ec(OpusCustomEncoder * OPUS_RESTRICT st, const opus_val16 * pcm, int frame_size, unsigned char *compressed, int nbCompressedBytes, ec_enc *enc);
137
138 int celt_encoder_init(CELTEncoder *st, opus_int32 sampling_rate, int channels,
139 int arch);
140
141
142
143 /* Decoder stuff */
144
145 int celt_decoder_get_size(int channels);
146
147
148 int celt_decoder_init(CELTDecoder *st, opus_int32 sampling_rate, int channels);
149
150 int celt_decode_with_ec(OpusCustomDecoder * OPUS_RESTRICT st, const unsigned char *data,
151 int len, opus_val16 * OPUS_RESTRICT pcm, int frame_size, ec_dec *dec, int accum);
152
153 #define celt_encoder_ctl opus_custom_encoder_ctl
154 #define celt_decoder_ctl opus_custom_decoder_ctl
155
156
157 #ifdef CUSTOM_MODES
158 #define OPUS_CUSTOM_NOSTATIC
159 #else
160 #define OPUS_CUSTOM_NOSTATIC static OPUS_INLINE
161 #endif
162
163 static const unsigned char trim_icdf[11] = {126, 124, 119, 109, 87, 41, 19, 9, 4, 2, 0};
164 /* Probs: NONE: 21.875%, LIGHT: 6.25%, NORMAL: 65.625%, AGGRESSIVE: 6.25% */
165 static const unsigned char spread_icdf[4] = {25, 23, 2, 0};
166
167 static const unsigned char tapset_icdf[3]={2,1,0};
168
169 #ifdef CUSTOM_MODES
170 static const unsigned char toOpusTable[20] = {
171 0xE0, 0xE8, 0xF0, 0xF8,
172 0xC0, 0xC8, 0xD0, 0xD8,
173 0xA0, 0xA8, 0xB0, 0xB8,
174 0x00, 0x00, 0x00, 0x00,
175 0x80, 0x88, 0x90, 0x98,
176 };
177
178 static const unsigned char fromOpusTable[16] = {
179 0x80, 0x88, 0x90, 0x98,
180 0x40, 0x48, 0x50, 0x58,
181 0x20, 0x28, 0x30, 0x38,
182 0x00, 0x08, 0x10, 0x18
183 };
184
toOpus(unsigned char c)185 static OPUS_INLINE int toOpus(unsigned char c)
186 {
187 int ret=0;
188 if (c<0xA0)
189 ret = toOpusTable[c>>3];
190 if (ret == 0)
191 return -1;
192 else
193 return ret|(c&0x7);
194 }
195
fromOpus(unsigned char c)196 static OPUS_INLINE int fromOpus(unsigned char c)
197 {
198 if (c<0x80)
199 return -1;
200 else
201 return fromOpusTable[(c>>3)-16] | (c&0x7);
202 }
203 #endif /* CUSTOM_MODES */
204
205 #define COMBFILTER_MAXPERIOD 1024
206 #define COMBFILTER_MINPERIOD 15
207
208 extern const signed char tf_select_table[4][8];
209
210 int resampling_factor(opus_int32 rate);
211
212 void celt_preemphasis(const opus_val16 * OPUS_RESTRICT pcmp, celt_sig * OPUS_RESTRICT inp,
213 int N, int CC, int upsample, const opus_val16 *coef, celt_sig *mem, int clip);
214
215 void comb_filter(opus_val32 *y, opus_val32 *x, int T0, int T1, int N,
216 opus_val16 g0, opus_val16 g1, int tapset0, int tapset1,
217 const opus_val16 *window, int overlap, int arch);
218
219 #ifdef NON_STATIC_COMB_FILTER_CONST_C
220 void comb_filter_const_c(opus_val32 *y, opus_val32 *x, int T, int N,
221 opus_val16 g10, opus_val16 g11, opus_val16 g12);
222 #endif
223
224 #ifndef OVERRIDE_COMB_FILTER_CONST
225 # define comb_filter_const(y, x, T, N, g10, g11, g12, arch) \
226 ((void)(arch),comb_filter_const_c(y, x, T, N, g10, g11, g12))
227 #endif
228
229 void init_caps(const CELTMode *m,int *cap,int LM,int C);
230
231 #ifdef RESYNTH
232 void deemphasis(celt_sig *in[], opus_val16 *pcm, int N, int C, int downsample, const opus_val16 *coef, celt_sig *mem);
233 void celt_synthesis(const CELTMode *mode, celt_norm *X, celt_sig * out_syn[],
234 opus_val16 *oldBandE, int start, int effEnd, int C, int CC, int isTransient,
235 int LM, int downsample, int silence);
236 #endif
237
238 #ifdef __cplusplus
239 }
240 #endif
241
242 #endif /* CELT_H */
243