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