ref: f1916a14fdb81543a7a5dc59aca77f357f8522a8
parent: 89039a3ff9997c93e790d57009214ffa42cdbfef
author: Jean-Marc Valin <[email protected]>
date: Mon Jan 31 05:51:30 EST 2011
Making it possible for the stereo decoder to decode a mono stream
--- a/libcelt/celt.c
+++ b/libcelt/celt.c
@@ -1734,6 +1734,7 @@
const CELTMode *mode;
int overlap;
int channels;
+ int stream_channels;
int downsample;
int start, end;
@@ -1832,7 +1833,7 @@
st->mode = mode;
st->overlap = mode->overlap;
- st->channels = channels;
+ st->stream_channels = st->channels = channels;
st->downsample = 1;
st->start = 0;
@@ -2102,7 +2103,7 @@
int shortBlocks;
int isTransient;
int intra_ener;
- const int C = CHANNELS(st->channels);
+ const int CC = CHANNELS(st->channels);
int LM, M;
int effEnd;
int codedBands;
@@ -2118,6 +2119,7 @@
int anti_collapse_rsv;
int anti_collapse_on=0;
int silence;
+ const int C = CHANNELS(st->stream_channels);
SAVE_STACK;
@@ -2136,12 +2138,12 @@
decode_mem[c] = st->_decode_mem + c*(DECODE_BUFFER_SIZE+st->overlap);
out_mem[c] = decode_mem[c]+DECODE_BUFFER_SIZE-MAX_PERIOD;
overlap_mem[c] = decode_mem[c]+DECODE_BUFFER_SIZE;
- } while (++c<C);
- lpc = (celt_word16*)(st->_decode_mem+(DECODE_BUFFER_SIZE+st->overlap)*C);
- oldBandE = lpc+C*LPC_ORDER;
- oldLogE = oldBandE + C*st->mode->nbEBands;
- oldLogE2 = oldLogE + C*st->mode->nbEBands;
- backgroundLogE = oldLogE2 + C*st->mode->nbEBands;
+ } while (++c<CC);
+ lpc = (celt_word16*)(st->_decode_mem+(DECODE_BUFFER_SIZE+st->overlap)*CC);
+ oldBandE = lpc+CC*LPC_ORDER;
+ oldLogE = oldBandE + CC*st->mode->nbEBands;
+ oldLogE2 = oldLogE + CC*st->mode->nbEBands;
+ backgroundLogE = oldLogE2 + CC*st->mode->nbEBands;
N = M*st->mode->shortMdctSize;
@@ -2149,17 +2151,17 @@
if (effEnd > st->mode->effEBands)
effEnd = st->mode->effEBands;
- ALLOC(freq, C*N, celt_sig); /**< Interleaved signal MDCTs */
- ALLOC(X, C*N, celt_norm); /**< Interleaved normalised MDCTs */
- ALLOC(bandE, st->mode->nbEBands*C, celt_ener);
+ ALLOC(freq, CC*N, celt_sig); /**< Interleaved signal MDCTs */
+ ALLOC(X, CC*N, celt_norm); /**< Interleaved normalised MDCTs */
+ ALLOC(bandE, st->mode->nbEBands*CC, celt_ener);
c=0; do
for (i=0;i<M*st->mode->eBands[st->start];i++)
X[c*N+i] = 0;
- while (++c<C);
+ while (++c<CC);
c=0; do
for (i=M*st->mode->eBands[effEnd];i<N;i++)
X[c*N+i] = 0;
- while (++c<C);
+ while (++c<CC);
if (data == NULL || len<=1)
{@@ -2179,6 +2181,21 @@
dec = &_dec;
}
+ if (CC==1&&C==2)
+ {+ RESTORE_STACK;
+ return CELT_CORRUPTED_DATA;
+ } else if (CC==2&&C==1)
+ {+ for (i=0;i<st->mode->nbEBands;i++)
+ {+ oldBandE[i]=MAX16(oldBandE[i],oldBandE[st->mode->nbEBands+i]);
+ oldLogE[i]=MAX16(oldLogE[i],oldLogE[st->mode->nbEBands+i]);
+ oldLogE2[i]=MAX16(oldLogE2[i],oldLogE2[st->mode->nbEBands+i]);
+ backgroundLogE[i]=MAX16(backgroundLogE[i],backgroundLogE[st->mode->nbEBands+i]);
+ }
+ }
+
total_bits = len*8;
tell = ec_dec_tell(dec, 0);
@@ -2327,7 +2344,7 @@
denormalise_bands(st->mode, X, freq, bandE, effEnd, C, M);
CELT_MOVE(decode_mem[0], decode_mem[0]+N, DECODE_BUFFER_SIZE-N);
- if (C==2)
+ if (CC==2)
CELT_MOVE(decode_mem[1], decode_mem[1]+N, DECODE_BUFFER_SIZE-N);
c=0; do
@@ -2343,11 +2360,17 @@
} while (++c<C);
out_syn[0] = out_mem[0]+MAX_PERIOD-N;
- if (C==2)
+ if (CC==2)
out_syn[1] = out_mem[1]+MAX_PERIOD-N;
+ if (CC==2&&C==1)
+ {+ for (i=0;i<N;i++)
+ freq[N+i] = freq[i];
+ }
+
/* Compute inverse MDCTs */
- compute_inv_mdcts(st->mode, shortBlocks, freq, out_syn, overlap_mem, C, LM);
+ compute_inv_mdcts(st->mode, shortBlocks, freq, out_syn, overlap_mem, CC, LM);
#ifdef ENABLE_POSTFILTER
c=0; do {@@ -2366,7 +2389,7 @@
st->postfilter_gain_old, st->postfilter_gain, st->postfilter_tapset_old, st->postfilter_tapset,
st->mode->window, st->mode->overlap);
}
- } while (++c<C);
+ } while (++c<CC);
st->postfilter_period_old = st->postfilter_period;
st->postfilter_gain_old = st->postfilter_gain;
st->postfilter_tapset_old = st->postfilter_tapset;
@@ -2375,6 +2398,11 @@
st->postfilter_tapset = postfilter_tapset;
#endif /* ENABLE_POSTFILTER */
+ if (CC==2&&C==1) {+ for (i=0;i<st->mode->nbEBands;i++)
+ oldBandE[st->mode->nbEBands+i]=oldBandE[i];
+ }
+
/* In case start or end were to change */
c=0; do
{@@ -2382,22 +2410,22 @@
oldBandE[c*st->mode->nbEBands+i]=0;
for (i=st->end;i<st->mode->nbEBands;i++)
oldBandE[c*st->mode->nbEBands+i]=0;
- } while (++c<C);
+ } while (++c<CC);
if (!isTransient)
{- for (i=0;i<C*st->mode->nbEBands;i++)
+ for (i=0;i<CC*st->mode->nbEBands;i++)
oldLogE2[i] = oldLogE[i];
- for (i=0;i<C*st->mode->nbEBands;i++)
+ for (i=0;i<CC*st->mode->nbEBands;i++)
oldLogE[i] = oldBandE[i];
- for (i=0;i<C*st->mode->nbEBands;i++)
+ for (i=0;i<CC*st->mode->nbEBands;i++)
backgroundLogE[i] = MIN16(backgroundLogE[i] + M*QCONST16(0.001f,DB_SHIFT), oldBandE[i]);
} else {- for (i=0;i<C*st->mode->nbEBands;i++)
+ for (i=0;i<CC*st->mode->nbEBands;i++)
oldLogE[i] = MIN16(oldLogE[i], oldBandE[i]);
}
st->rng = dec->rng;
- deemphasis(out_syn, pcm, N, C, st->downsample, st->mode->preemph, st->preemph_memD);
+ deemphasis(out_syn, pcm, N, CC, st->downsample, st->mode->preemph, st->preemph_memD);
st->loss_count = 0;
RESTORE_STACK;
if (ec_dec_tell(dec,0) > 8*len || ec_dec_get_error(dec))
@@ -2510,6 +2538,14 @@
if (value<0 || value>=st->mode->nbEBands)
goto bad_arg;
st->end = value;
+ }
+ break;
+ case CELT_SET_CHANNELS_REQUEST:
+ {+ celt_int32 value = va_arg(ap, celt_int32);
+ if (value<1 || value>2)
+ goto bad_arg;
+ st->stream_channels = value;
}
break;
case CELT_RESET_STATE:
--- a/libcelt/celt.h
+++ b/libcelt/celt.h
@@ -104,6 +104,9 @@
#define CELT_SET_END_BAND_REQUEST 10001
#define CELT_SET_END_BAND(x) CELT_SET_END_BAND_REQUEST, _celt_check_int(x)
+#define CELT_SET_CHANNELS_REQUEST 10002
+#define CELT_SET_CHANNELS(x) CELT_SET_CHANNELS_REQUEST, _celt_check_int(x)
+
/** GET the lookahead used in the current mode */
#define CELT_GET_LOOKAHEAD 1001
/** GET the sample rate used in the current mode */