Commit 5e7c422d authored by Justin Ruggles's avatar Justin Ruggles Committed by Michael Niedermayer

vmdaudio: add out_bps to VmdAudioContext and use it to replace hard-coded sample size.

Signed-off-by: 's avatarRonald S. Bultje <rsbultje@gmail.com>
(cherry picked from commit 1e86d685)
parent 504dff8e
...@@ -420,6 +420,7 @@ static av_cold int vmdvideo_decode_end(AVCodecContext *avctx) ...@@ -420,6 +420,7 @@ static av_cold int vmdvideo_decode_end(AVCodecContext *avctx)
typedef struct VmdAudioContext { typedef struct VmdAudioContext {
AVCodecContext *avctx; AVCodecContext *avctx;
int out_bps;
int channels; int channels;
int bits; int bits;
int block_align; int block_align;
...@@ -451,6 +452,7 @@ static av_cold int vmdaudio_decode_init(AVCodecContext *avctx) ...@@ -451,6 +452,7 @@ static av_cold int vmdaudio_decode_init(AVCodecContext *avctx)
s->bits = avctx->bits_per_coded_sample; s->bits = avctx->bits_per_coded_sample;
s->block_align = avctx->block_align; s->block_align = avctx->block_align;
avctx->sample_fmt = AV_SAMPLE_FMT_S16; avctx->sample_fmt = AV_SAMPLE_FMT_S16;
s->out_bps = av_get_bits_per_sample_fmt(avctx->sample_fmt) >> 3;
av_log(s->avctx, AV_LOG_DEBUG, "%d channels, %d bits/sample, block align = %d, sample rate = %d\n", av_log(s->avctx, AV_LOG_DEBUG, "%d channels, %d bits/sample, block align = %d, sample rate = %d\n",
s->channels, s->bits, s->block_align, avctx->sample_rate); s->channels, s->bits, s->block_align, avctx->sample_rate);
...@@ -480,7 +482,7 @@ static int vmdaudio_loadsound(VmdAudioContext *s, unsigned char *data, ...@@ -480,7 +482,7 @@ static int vmdaudio_loadsound(VmdAudioContext *s, unsigned char *data,
const uint8_t *buf, int silent_chunks, int data_size) const uint8_t *buf, int silent_chunks, int data_size)
{ {
int i; int i;
int silent_size = s->block_align * silent_chunks * 2; int silent_size = s->block_align * silent_chunks * s->out_bps;
if (silent_chunks) { if (silent_chunks) {
memset(data, 0, silent_size); memset(data, 0, silent_size);
...@@ -496,7 +498,7 @@ static int vmdaudio_loadsound(VmdAudioContext *s, unsigned char *data, ...@@ -496,7 +498,7 @@ static int vmdaudio_loadsound(VmdAudioContext *s, unsigned char *data,
} }
} }
return silent_size + data_size * 2; return silent_size + data_size * s->out_bps;
} }
static int vmdaudio_decode_frame(AVCodecContext *avctx, static int vmdaudio_decode_frame(AVCodecContext *avctx,
...@@ -535,7 +537,7 @@ static int vmdaudio_decode_frame(AVCodecContext *avctx, ...@@ -535,7 +537,7 @@ static int vmdaudio_decode_frame(AVCodecContext *avctx,
} }
/* ensure output buffer is large enough */ /* ensure output buffer is large enough */
if (*data_size < (s->block_align*silent_chunks + buf_size) * 2) if (*data_size < (s->block_align*silent_chunks + buf_size) * s->out_bps)
return -1; return -1;
*data_size = vmdaudio_loadsound(s, output_samples, buf, silent_chunks, buf_size); *data_size = vmdaudio_loadsound(s, output_samples, buf, silent_chunks, buf_size);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment