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
46extern "C" {
47#endif
48
49#define CELTEncoder OpusCustomEncoder
50#define CELTDecoder OpusCustomDecoder
51#define CELTMode OpusCustomMode
52
53typedef struct {
54   int valid;
55   float tonality;
56   float tonality_slope;
57   float noisiness;
58   float activity;
59   float music_prob;
60   int        bandwidth;
61}AnalysisInfo;
62
63#define __celt_check_mode_ptr_ptr(ptr) ((ptr) + ((ptr) - (const CELTMode**)(ptr)))
64
65#define __celt_check_analysis_ptr(ptr) ((ptr) + ((ptr) - (const AnalysisInfo*)(ptr)))
66
67/* Encoder/decoder Requests */
68
69/* Expose this option again when variable framesize actually works */
70#define OPUS_FRAMESIZE_VARIABLE              5010 /**< Optimize the frame size dynamically */
71
72
73#define CELT_SET_PREDICTION_REQUEST    10002
74/** Controls the use of interframe prediction.
75    0=Independent frames
76    1=Short term interframe prediction allowed
77    2=Long term prediction allowed
78 */
79#define CELT_SET_PREDICTION(x) CELT_SET_PREDICTION_REQUEST, __opus_check_int(x)
80
81#define CELT_SET_INPUT_CLIPPING_REQUEST    10004
82#define CELT_SET_INPUT_CLIPPING(x) CELT_SET_INPUT_CLIPPING_REQUEST, __opus_check_int(x)
83
84#define CELT_GET_AND_CLEAR_ERROR_REQUEST   10007
85#define CELT_GET_AND_CLEAR_ERROR(x) CELT_GET_AND_CLEAR_ERROR_REQUEST, __opus_check_int_ptr(x)
86
87#define CELT_SET_CHANNELS_REQUEST    10008
88#define CELT_SET_CHANNELS(x) CELT_SET_CHANNELS_REQUEST, __opus_check_int(x)
89
90
91/* Internal */
92#define CELT_SET_START_BAND_REQUEST    10010
93#define CELT_SET_START_BAND(x) CELT_SET_START_BAND_REQUEST, __opus_check_int(x)
94
95#define CELT_SET_END_BAND_REQUEST    10012
96#define CELT_SET_END_BAND(x) CELT_SET_END_BAND_REQUEST, __opus_check_int(x)
97
98#define CELT_GET_MODE_REQUEST    10015
99/** Get the CELTMode used by an encoder or decoder */
100#define CELT_GET_MODE(x) CELT_GET_MODE_REQUEST, __celt_check_mode_ptr_ptr(x)
101
102#define CELT_SET_SIGNALLING_REQUEST    10016
103#define CELT_SET_SIGNALLING(x) CELT_SET_SIGNALLING_REQUEST, __opus_check_int(x)
104
105#define CELT_SET_TONALITY_REQUEST    10018
106#define CELT_SET_TONALITY(x) CELT_SET_TONALITY_REQUEST, __opus_check_int(x)
107#define CELT_SET_TONALITY_SLOPE_REQUEST    10020
108#define CELT_SET_TONALITY_SLOPE(x) CELT_SET_TONALITY_SLOPE_REQUEST, __opus_check_int(x)
109
110#define CELT_SET_ANALYSIS_REQUEST    10022
111#define CELT_SET_ANALYSIS(x) CELT_SET_ANALYSIS_REQUEST, __celt_check_analysis_ptr(x)
112
113#define OPUS_SET_LFE_REQUEST    10024
114#define OPUS_SET_LFE(x) OPUS_SET_LFE_REQUEST, __opus_check_int(x)
115
116#define OPUS_SET_ENERGY_MASK_REQUEST    10026
117#define OPUS_SET_ENERGY_MASK(x) OPUS_SET_ENERGY_MASK_REQUEST, __opus_check_val16_ptr(x)
118
119/* Encoder stuff */
120
121int celt_encoder_get_size(int channels);
122
123int celt_encode_with_ec(OpusCustomEncoder * OPUS_RESTRICT st, const opus_val16 * pcm, int frame_size, unsigned char *compressed, int nbCompressedBytes, ec_enc *enc);
124
125int celt_encoder_init(CELTEncoder *st, opus_int32 sampling_rate, int channels,
126                      int arch);
127
128
129
130/* Decoder stuff */
131
132int celt_decoder_get_size(int channels);
133
134
135int celt_decoder_init(CELTDecoder *st, opus_int32 sampling_rate, int channels);
136
137int celt_decode_with_ec(OpusCustomDecoder * OPUS_RESTRICT st, const unsigned char *data, int len, opus_val16 * OPUS_RESTRICT pcm, int frame_size, ec_dec *dec);
138
139#define celt_encoder_ctl opus_custom_encoder_ctl
140#define celt_decoder_ctl opus_custom_decoder_ctl
141
142
143#ifdef CUSTOM_MODES
144#define OPUS_CUSTOM_NOSTATIC
145#else
146#define OPUS_CUSTOM_NOSTATIC static OPUS_INLINE
147#endif
148
149static const unsigned char trim_icdf[11] = {126, 124, 119, 109, 87, 41, 19, 9, 4, 2, 0};
150/* Probs: NONE: 21.875%, LIGHT: 6.25%, NORMAL: 65.625%, AGGRESSIVE: 6.25% */
151static const unsigned char spread_icdf[4] = {25, 23, 2, 0};
152
153static const unsigned char tapset_icdf[3]={2,1,0};
154
155#ifdef CUSTOM_MODES
156static const unsigned char toOpusTable[20] = {
157      0xE0, 0xE8, 0xF0, 0xF8,
158      0xC0, 0xC8, 0xD0, 0xD8,
159      0xA0, 0xA8, 0xB0, 0xB8,
160      0x00, 0x00, 0x00, 0x00,
161      0x80, 0x88, 0x90, 0x98,
162};
163
164static const unsigned char fromOpusTable[16] = {
165      0x80, 0x88, 0x90, 0x98,
166      0x40, 0x48, 0x50, 0x58,
167      0x20, 0x28, 0x30, 0x38,
168      0x00, 0x08, 0x10, 0x18
169};
170
171static OPUS_INLINE int toOpus(unsigned char c)
172{
173   int ret=0;
174   if (c<0xA0)
175      ret = toOpusTable[c>>3];
176   if (ret == 0)
177      return -1;
178   else
179      return ret|(c&0x7);
180}
181
182static OPUS_INLINE int fromOpus(unsigned char c)
183{
184   if (c<0x80)
185      return -1;
186   else
187      return fromOpusTable[(c>>3)-16] | (c&0x7);
188}
189#endif /* CUSTOM_MODES */
190
191#define COMBFILTER_MAXPERIOD 1024
192#define COMBFILTER_MINPERIOD 15
193
194extern const signed char tf_select_table[4][8];
195
196int resampling_factor(opus_int32 rate);
197
198void celt_preemphasis(const opus_val16 * OPUS_RESTRICT pcmp, celt_sig * OPUS_RESTRICT inp,
199                        int N, int CC, int upsample, const opus_val16 *coef, celt_sig *mem, int clip);
200
201void comb_filter(opus_val32 *y, opus_val32 *x, int T0, int T1, int N,
202      opus_val16 g0, opus_val16 g1, int tapset0, int tapset1,
203      const opus_val16 *window, int overlap);
204
205void init_caps(const CELTMode *m,int *cap,int LM,int C);
206
207#ifdef RESYNTH
208void deemphasis(celt_sig *in[], opus_val16 *pcm, int N, int C, int downsample, const opus_val16 *coef, celt_sig *mem, celt_sig * OPUS_RESTRICT scratch);
209
210void compute_inv_mdcts(const CELTMode *mode, int shortBlocks, celt_sig *X,
211      celt_sig * OPUS_RESTRICT out_mem[], int C, int LM);
212#endif
213
214#ifdef __cplusplus
215}
216#endif
217
218#endif /* CELT_H */
219