Index: src/opus_encoder.c |
diff --git a/src/opus_encoder.c b/src/opus_encoder.c |
index f6f0ce5c3ad162e7a0fa67ed28d98b85b8549f1d..fbd3de6392159ab5e4e75b41154b93a8764c3bd4 100644 |
--- a/src/opus_encoder.c |
+++ b/src/opus_encoder.c |
@@ -104,7 +104,7 @@ struct OpusEncoder { |
int analysis_offset; |
#endif |
opus_uint32 rangeFinal; |
- int arch; |
+ int arch; |
}; |
/* Transition tables for the voice and music. First column is the |
@@ -114,35 +114,35 @@ static const opus_int32 mono_voice_bandwidth_thresholds[8] = { |
11000, 1000, /* NB<->MB */ |
14000, 1000, /* MB<->WB */ |
17000, 1000, /* WB<->SWB */ |
- 20000, 1000, /* SWB<->FB */ |
+ 21000, 2000, /* SWB<->FB */ |
}; |
static const opus_int32 mono_music_bandwidth_thresholds[8] = { |
- 14000, 1000, /* MB not allowed */ |
- 18000, 2000, /* MB<->WB */ |
- 24000, 2000, /* WB<->SWB */ |
- 33000, 2000, /* SWB<->FB */ |
+ 12000, 1000, /* NB<->MB */ |
+ 15000, 1000, /* MB<->WB */ |
+ 18000, 2000, /* WB<->SWB */ |
+ 22000, 2000, /* SWB<->FB */ |
}; |
static const opus_int32 stereo_voice_bandwidth_thresholds[8] = { |
11000, 1000, /* NB<->MB */ |
14000, 1000, /* MB<->WB */ |
21000, 2000, /* WB<->SWB */ |
- 32000, 2000, /* SWB<->FB */ |
+ 28000, 2000, /* SWB<->FB */ |
}; |
static const opus_int32 stereo_music_bandwidth_thresholds[8] = { |
- 14000, 1000, /* MB not allowed */ |
+ 12000, 1000, /* NB<->MB */ |
18000, 2000, /* MB<->WB */ |
- 24000, 2000, /* WB<->SWB */ |
- 48000, 2000, /* SWB<->FB */ |
+ 21000, 2000, /* WB<->SWB */ |
+ 30000, 2000, /* SWB<->FB */ |
}; |
/* Threshold bit-rates for switching between mono and stereo */ |
-static const opus_int32 stereo_voice_threshold = 31000; |
-static const opus_int32 stereo_music_threshold = 31000; |
+static const opus_int32 stereo_voice_threshold = 30000; |
+static const opus_int32 stereo_music_threshold = 30000; |
/* Threshold bit-rate for switching between SILK/hybrid and CELT-only */ |
static const opus_int32 mode_thresholds[2][2] = { |
/* voice */ /* music */ |
- { 64000, 20000}, /* mono */ |
- { 36000, 20000}, /* stereo */ |
+ { 64000, 16000}, /* mono */ |
+ { 36000, 16000}, /* stereo */ |
}; |
int opus_encoder_get_size(int channels) |
@@ -188,7 +188,7 @@ int opus_encoder_init(OpusEncoder* st, opus_int32 Fs, int channels, int applicat |
st->arch = opus_select_arch(); |
- ret = silk_InitEncoder( silk_enc, &st->silk_mode ); |
+ ret = silk_InitEncoder( silk_enc, st->arch, &st->silk_mode ); |
if(ret)return OPUS_INTERNAL_ERROR; |
/* default SILK parameters */ |
@@ -205,10 +205,11 @@ int opus_encoder_init(OpusEncoder* st, opus_int32 Fs, int channels, int applicat |
st->silk_mode.useInBandFEC = 0; |
st->silk_mode.useDTX = 0; |
st->silk_mode.useCBR = 0; |
+ st->silk_mode.reducedDependency = 0; |
/* Create CELT encoder */ |
/* Initialize CELT encoder */ |
- err = celt_encoder_init(celt_enc, Fs, channels); |
+ err = celt_encoder_init(celt_enc, Fs, channels, st->arch); |
if(err!=OPUS_OK)return OPUS_INTERNAL_ERROR; |
celt_encoder_ctl(celt_enc, CELT_SET_SIGNALLING(0)); |
@@ -244,44 +245,6 @@ int opus_encoder_init(OpusEncoder* st, opus_int32 Fs, int channels, int applicat |
return OPUS_OK; |
} |
-static int pad_frame(unsigned char *data, opus_int32 len, opus_int32 new_len) |
-{ |
- if (len == new_len) |
- return 0; |
- if (len > new_len) |
- return 1; |
- |
- if ((data[0]&0x3)==0) |
- { |
- int i; |
- int padding, nb_255s; |
- |
- padding = new_len - len; |
- if (padding >= 2) |
- { |
- nb_255s = (padding-2)/255; |
- |
- for (i=len-1;i>=1;i--) |
- data[i+nb_255s+2] = data[i]; |
- data[0] |= 0x3; |
- data[1] = 0x41; |
- for (i=0;i<nb_255s;i++) |
- data[i+2] = 255; |
- data[nb_255s+2] = padding-255*nb_255s-2; |
- for (i=len+3+nb_255s;i<new_len;i++) |
- data[i] = 0; |
- } else { |
- for (i=len-1;i>=1;i--) |
- data[i+1] = data[i]; |
- data[0] |= 0x3; |
- data[1] = 1; |
- } |
- return 0; |
- } else { |
- return 1; |
- } |
-} |
- |
static unsigned char gen_toc(int mode, int framerate, int bandwidth, int channels) |
{ |
int period; |
@@ -686,14 +649,14 @@ static int transient_viterbi(const float *E, const float *E_1, int N, int frame_ |
} |
int optimize_framesize(const opus_val16 *x, int len, int C, opus_int32 Fs, |
- int bitrate, opus_val16 tonality, opus_val32 *mem, int buffering, |
+ int bitrate, opus_val16 tonality, float *mem, int buffering, |
downmix_func downmix) |
{ |
int N; |
int i; |
float e[MAX_DYNAMIC_FRAMESIZE+4]; |
float e_1[MAX_DYNAMIC_FRAMESIZE+3]; |
- float memx; |
+ opus_val32 memx; |
int bestLM=0; |
int subframe; |
int pos; |
@@ -720,11 +683,12 @@ int optimize_framesize(const opus_val16 *x, int len, int C, opus_int32 Fs, |
pos=1; |
} |
N=IMIN(len/subframe, MAX_DYNAMIC_FRAMESIZE); |
- memx = x[0]; |
+ /* Just silencing a warning, it's really initialized later */ |
+ memx = 0; |
for (i=0;i<N;i++) |
{ |
float tmp; |
- float tmpx; |
+ opus_val32 tmpx; |
int j; |
tmp=EPSILON; |
@@ -734,7 +698,7 @@ int optimize_framesize(const opus_val16 *x, int len, int C, opus_int32 Fs, |
for (j=0;j<subframe;j++) |
{ |
tmpx = sub[j]; |
- tmp += (tmpx-memx)*(tmpx-memx); |
+ tmp += (tmpx-memx)*(float)(tmpx-memx); |
memx = tmpx; |
} |
e[i+pos] = tmp; |
@@ -856,7 +820,11 @@ opus_int32 frame_size_select(opus_int32 frame_size, int variable_duration, opus_ |
opus_int32 compute_frame_size(const void *analysis_pcm, int frame_size, |
int variable_duration, int C, opus_int32 Fs, int bitrate_bps, |
- int delay_compensation, downmix_func downmix, opus_val32 *subframe_mem) |
+ int delay_compensation, downmix_func downmix |
+#ifndef DISABLE_FLOAT_API |
+ , float *subframe_mem |
+#endif |
+ ) |
{ |
#ifndef DISABLE_FLOAT_API |
if (variable_duration == OPUS_FRAMESIZE_VARIABLE && frame_size >= Fs/200) |
@@ -1095,7 +1063,7 @@ opus_int32 opus_encode_native(OpusEncoder *st, const opus_val16 *pcm, int frame_ |
max_rate = frame_rate*max_data_bytes*8; |
/* Equivalent 20-ms rate for mode/channel/bandwidth decisions */ |
- equiv_rate = st->bitrate_bps - 60*(st->Fs/frame_size - 50); |
+ equiv_rate = st->bitrate_bps - (40*st->channels+20)*(st->Fs/frame_size - 50); |
if (st->signal_type == OPUS_SIGNAL_VOICE) |
voice_est = 127; |
@@ -1136,6 +1104,7 @@ opus_int32 opus_encode_native(OpusEncoder *st, const opus_val16 *pcm, int frame_ |
} |
#endif |
} |
+ equiv_rate = st->bitrate_bps - (40*st->stream_channels+20)*(st->Fs/frame_size - 50); |
/* Mode selection depending on application and signal type */ |
if (st->application == OPUS_APPLICATION_RESTRICTED_LOWDELAY) |
@@ -1195,6 +1164,11 @@ opus_int32 opus_encode_native(OpusEncoder *st, const opus_val16 *pcm, int frame_ |
/* Override the chosen mode to make sure we meet the requested frame size */ |
if (st->mode != MODE_CELT_ONLY && frame_size < st->Fs/100) |
st->mode = MODE_CELT_ONLY; |
+ if (st->lfe) |
+ st->mode = MODE_CELT_ONLY; |
+ /* If max_data_bytes represents less than 8 kb/s, switch to CELT-only mode */ |
+ if (max_data_bytes < (frame_rate > 50 ? 12000 : 8000)*frame_size / (st->Fs * 8)) |
+ st->mode = MODE_CELT_ONLY; |
if (st->stream_channels == 1 && st->prev_channels ==2 && st->silk_mode.toMono==0 |
&& st->mode != MODE_CELT_ONLY && st->prev_mode != MODE_CELT_ONLY) |
@@ -1245,7 +1219,7 @@ opus_int32 opus_encode_native(OpusEncoder *st, const opus_val16 *pcm, int frame_ |
if (st->mode != MODE_CELT_ONLY && st->prev_mode == MODE_CELT_ONLY) |
{ |
silk_EncControlStruct dummy; |
- silk_InitEncoder( silk_enc, &dummy); |
+ silk_InitEncoder( silk_enc, st->arch, &dummy); |
prefill=1; |
} |
@@ -1323,7 +1297,7 @@ opus_int32 opus_encode_native(OpusEncoder *st, const opus_val16 *pcm, int frame_ |
st->bandwidth = OPUS_BANDWIDTH_MEDIUMBAND; |
if (st->Fs <= 8000 && st->bandwidth > OPUS_BANDWIDTH_NARROWBAND) |
st->bandwidth = OPUS_BANDWIDTH_NARROWBAND; |
-#ifndef FIXED_POINT |
+#ifndef DISABLE_FLOAT_API |
/* Use detected bandwidth to reduce the encoded bandwidth. */ |
if (st->detected_bandwidth && st->user_bandwidth == OPUS_AUTO) |
{ |
@@ -1332,13 +1306,13 @@ opus_int32 opus_encode_native(OpusEncoder *st, const opus_val16 *pcm, int frame_ |
gets it wrong when we could have coded a high bandwidth transparently. |
When operating in SILK/hybrid mode, we don't go below wideband to avoid |
more complicated switches that require redundancy. */ |
- if (st->bitrate_bps <= 18000*st->stream_channels && st->mode == MODE_CELT_ONLY) |
+ if (equiv_rate <= 18000*st->stream_channels && st->mode == MODE_CELT_ONLY) |
min_detected_bandwidth = OPUS_BANDWIDTH_NARROWBAND; |
- else if (st->bitrate_bps <= 24000*st->stream_channels && st->mode == MODE_CELT_ONLY) |
+ else if (equiv_rate <= 24000*st->stream_channels && st->mode == MODE_CELT_ONLY) |
min_detected_bandwidth = OPUS_BANDWIDTH_MEDIUMBAND; |
- else if (st->bitrate_bps <= 30000*st->stream_channels) |
+ else if (equiv_rate <= 30000*st->stream_channels) |
min_detected_bandwidth = OPUS_BANDWIDTH_WIDEBAND; |
- else if (st->bitrate_bps <= 44000*st->stream_channels) |
+ else if (equiv_rate <= 44000*st->stream_channels) |
min_detected_bandwidth = OPUS_BANDWIDTH_SUPERWIDEBAND; |
else |
min_detected_bandwidth = OPUS_BANDWIDTH_FULLBAND; |
@@ -1349,18 +1323,11 @@ opus_int32 opus_encode_native(OpusEncoder *st, const opus_val16 *pcm, int frame_ |
#endif |
celt_encoder_ctl(celt_enc, OPUS_SET_LSB_DEPTH(lsb_depth)); |
- /* If max_data_bytes represents less than 8 kb/s, switch to CELT-only mode */ |
- if (max_data_bytes < (frame_rate > 50 ? 12000 : 8000)*frame_size / (st->Fs * 8)) |
- st->mode = MODE_CELT_ONLY; |
- |
/* CELT mode doesn't support mediumband, use wideband instead */ |
if (st->mode == MODE_CELT_ONLY && st->bandwidth == OPUS_BANDWIDTH_MEDIUMBAND) |
st->bandwidth = OPUS_BANDWIDTH_WIDEBAND; |
if (st->lfe) |
- { |
st->bandwidth = OPUS_BANDWIDTH_NARROWBAND; |
- st->mode = MODE_CELT_ONLY; |
- } |
/* Can't support higher than wideband for >20 ms frames */ |
if (frame_size > st->Fs/50 && (st->mode == MODE_CELT_ONLY || st->bandwidth > OPUS_BANDWIDTH_WIDEBAND)) |
@@ -1370,12 +1337,15 @@ opus_int32 opus_encode_native(OpusEncoder *st, const opus_val16 *pcm, int frame_ |
int bak_mode, bak_bandwidth, bak_channels, bak_to_mono; |
VARDECL(OpusRepacketizer, rp); |
opus_int32 bytes_per_frame; |
+ opus_int32 repacketize_len; |
+#ifndef DISABLE_FLOAT_API |
if (analysis_read_pos_bak!= -1) |
{ |
st->analysis.read_pos = analysis_read_pos_bak; |
st->analysis.read_subframe = analysis_read_subframe_bak; |
} |
+#endif |
nb_frames = frame_size > st->Fs/25 ? 3 : 2; |
bytes_per_frame = IMIN(1276,(out_data_bytes-3)/nb_frames); |
@@ -1420,7 +1390,11 @@ opus_int32 opus_encode_native(OpusEncoder *st, const opus_val16 *pcm, int frame_ |
return OPUS_INTERNAL_ERROR; |
} |
} |
- ret = opus_repacketizer_out(rp, data, out_data_bytes); |
+ if (st->use_vbr) |
+ repacketize_len = out_data_bytes; |
+ else |
+ repacketize_len = IMIN(3*st->bitrate_bps/(3*8*50/nb_frames), out_data_bytes); |
+ ret = opus_repacketizer_out_range_impl(rp, 0, nb_frames, data, repacketize_len, 0, !st->use_vbr); |
if (ret<0) |
{ |
RESTORE_STACK; |
@@ -1502,11 +1476,14 @@ opus_int32 opus_encode_native(OpusEncoder *st, const opus_val16 *pcm, int frame_ |
if( st->silk_mode.bitRate > total_bitRate * 4/5 ) { |
st->silk_mode.bitRate = total_bitRate * 4/5; |
} |
- /* Increasingly attenuate high band when it gets allocated fewer bits */ |
- celt_rate = total_bitRate - st->silk_mode.bitRate; |
- HB_gain_ref = (curr_bandwidth == OPUS_BANDWIDTH_SUPERWIDEBAND) ? 3000 : 3600; |
- HB_gain = SHL32((opus_val32)celt_rate, 9) / SHR32((opus_val32)celt_rate + st->stream_channels * HB_gain_ref, 6); |
- HB_gain = HB_gain < Q15ONE*6/7 ? HB_gain + Q15ONE/7 : Q15ONE; |
+ if (!st->energy_masking) |
+ { |
+ /* Increasingly attenuate high band when it gets allocated fewer bits */ |
+ celt_rate = total_bitRate - st->silk_mode.bitRate; |
+ HB_gain_ref = (curr_bandwidth == OPUS_BANDWIDTH_SUPERWIDEBAND) ? 3000 : 3600; |
+ HB_gain = SHL32((opus_val32)celt_rate, 9) / SHR32((opus_val32)celt_rate + st->stream_channels * HB_gain_ref, 6); |
+ HB_gain = HB_gain < Q15ONE*6/7 ? HB_gain + Q15ONE/7 : Q15ONE; |
+ } |
} else { |
/* SILK gets all bits */ |
st->silk_mode.bitRate = total_bitRate; |
@@ -1536,18 +1513,22 @@ opus_int32 opus_encode_native(OpusEncoder *st, const opus_val16 *pcm, int frame_ |
{ |
opus_val16 mask; |
mask = MAX16(MIN16(st->energy_masking[21*c+i], |
- QCONST16(.25f, DB_SHIFT)), -QCONST16(2.0f, DB_SHIFT)); |
+ QCONST16(.5f, DB_SHIFT)), -QCONST16(2.0f, DB_SHIFT)); |
if (mask > 0) |
mask = HALF16(mask); |
mask_sum += mask; |
} |
} |
/* Conservative rate reduction, we cut the masking in half */ |
- masking_depth = HALF16(mask_sum / end*st->channels); |
- rate_offset = PSHR32(MULT16_16(srate, masking_depth), DB_SHIFT); |
+ masking_depth = mask_sum / end*st->channels; |
+ masking_depth += QCONST16(.2f, DB_SHIFT); |
+ rate_offset = (opus_int32)PSHR32(MULT16_16(srate, masking_depth), DB_SHIFT); |
rate_offset = MAX32(rate_offset, -2*st->silk_mode.bitRate/3); |
- rate_offset += QCONST16(.4f, DB_SHIFT); |
- st->silk_mode.bitRate += rate_offset; |
+ /* Split the rate change between the SILK and CELT part for hybrid. */ |
+ if (st->bandwidth==OPUS_BANDWIDTH_SUPERWIDEBAND || st->bandwidth==OPUS_BANDWIDTH_FULLBAND) |
+ st->silk_mode.bitRate += 3*rate_offset/5; |
+ else |
+ st->silk_mode.bitRate += rate_offset; |
bytes_target += rate_offset * frame_size / (8 * st->Fs); |
} |
@@ -1698,9 +1679,12 @@ opus_int32 opus_encode_native(OpusEncoder *st, const opus_val16 *pcm, int frame_ |
celt_encoder_ctl(celt_enc, OPUS_SET_BITRATE(OPUS_BITRATE_MAX)); |
if (st->mode != MODE_SILK_ONLY) |
{ |
+ opus_val32 celt_pred=2; |
celt_encoder_ctl(celt_enc, OPUS_SET_VBR(0)); |
- /* Allow prediction unless we decide to disable it later */ |
- celt_encoder_ctl(celt_enc, CELT_SET_PREDICTION(2)); |
+ /* We may still decide to disable prediction later */ |
+ if (st->silk_mode.reducedDependency) |
+ celt_pred = 0; |
+ celt_encoder_ctl(celt_enc, CELT_SET_PREDICTION(celt_pred)); |
if (st->mode == MODE_HYBRID) |
{ |
@@ -1760,7 +1744,7 @@ opus_int32 opus_encode_native(OpusEncoder *st, const opus_val16 *pcm, int frame_ |
} |
st->prev_HB_gain = HB_gain; |
if (st->mode != MODE_HYBRID || st->stream_channels==1) |
- st->silk_mode.stereoWidth_Q14 = IMIN((1<<14),IMAX(0,st->bitrate_bps-32000)); |
+ st->silk_mode.stereoWidth_Q14 = IMIN((1<<14),2*IMAX(0,equiv_rate-30000)); |
if( !st->energy_masking && st->channels == 2 ) { |
/* Apply stereo width reduction (at low bitrates) */ |
if( st->hybrid_stereo_width_Q14 < (1 << 14) || st->silk_mode.stereoWidth_Q14 < (1 << 14) ) { |
@@ -1790,7 +1774,7 @@ opus_int32 opus_encode_native(OpusEncoder *st, const opus_val16 *pcm, int frame_ |
int max_redundancy; |
ec_enc_bit_logp(&enc, celt_to_silk, 1); |
if (st->mode == MODE_HYBRID) |
- max_redundancy = (max_data_bytes-1)-nb_compr_bytes-1; |
+ max_redundancy = (max_data_bytes-1)-nb_compr_bytes; |
else |
max_redundancy = (max_data_bytes-1)-((ec_tell(&enc)+7)>>3); |
/* Target the same bit-rate for redundancy as for the rest, |
@@ -1933,9 +1917,10 @@ opus_int32 opus_encode_native(OpusEncoder *st, const opus_val16 *pcm, int frame_ |
} |
/* Count ToC and redundancy */ |
ret += 1+redundancy_bytes; |
- if (!st->use_vbr && ret >= 3) |
+ if (!st->use_vbr) |
{ |
- if (pad_frame(data, ret, max_data_bytes)) |
+ if (opus_packet_pad(data, ret, max_data_bytes) != OPUS_OK) |
+ |
{ |
RESTORE_STACK; |
return OPUS_INTERNAL_ERROR; |
@@ -1987,7 +1972,11 @@ opus_int32 opus_encode(OpusEncoder *st, const opus_int16 *pcm, int analysis_fram |
delay_compensation = st->delay_compensation; |
frame_size = compute_frame_size(pcm, analysis_frame_size, |
st->variable_duration, st->channels, st->Fs, st->bitrate_bps, |
- delay_compensation, downmix_float, st->analysis.subframe_mem); |
+ delay_compensation, downmix_int |
+#ifndef DISABLE_FLOAT_API |
+ , st->analysis.subframe_mem |
+#endif |
+ ); |
return opus_encode_native(st, pcm, frame_size, data, out_data_bytes, 16, pcm, analysis_frame_size, 0, -2, st->channels, downmix_int); |
} |
@@ -2007,7 +1996,7 @@ opus_int32 opus_encode(OpusEncoder *st, const opus_int16 *pcm, int analysis_fram |
delay_compensation = st->delay_compensation; |
frame_size = compute_frame_size(pcm, analysis_frame_size, |
st->variable_duration, st->channels, st->Fs, st->bitrate_bps, |
- delay_compensation, downmix_float, st->analysis.subframe_mem); |
+ delay_compensation, downmix_int, st->analysis.subframe_mem); |
ALLOC(in, frame_size*st->channels, float); |
@@ -2409,6 +2398,22 @@ int opus_encoder_ctl(OpusEncoder *st, int request, ...) |
*value = st->variable_duration; |
} |
break; |
+ case OPUS_SET_PREDICTION_DISABLED_REQUEST: |
+ { |
+ opus_int32 value = va_arg(ap, opus_int32); |
+ if (value > 1 || value < 0) |
+ goto bad_arg; |
+ st->silk_mode.reducedDependency = value; |
+ } |
+ break; |
+ case OPUS_GET_PREDICTION_DISABLED_REQUEST: |
+ { |
+ opus_int32 *value = va_arg(ap, opus_int32*); |
+ if (!value) |
+ goto bad_arg; |
+ *value = st->silk_mode.reducedDependency; |
+ } |
+ break; |
case OPUS_RESET_STATE: |
{ |
void *silk_enc; |
@@ -2420,7 +2425,7 @@ int opus_encoder_ctl(OpusEncoder *st, int request, ...) |
((char*)&st->OPUS_ENCODER_RESET_START - (char*)st)); |
celt_encoder_ctl(celt_enc, OPUS_RESET_STATE); |
- silk_InitEncoder( silk_enc, &dummy ); |
+ silk_InitEncoder( silk_enc, st->arch, &dummy ); |
st->stream_channels = st->channels; |
st->hybrid_stereo_width_Q14 = 1 << 14; |
st->prev_HB_gain = Q15ONE; |