Commit ad2104ba authored by Justin Ruggles's avatar Justin Ruggles

tta: decode directly to the user-provided AVFrame

parent 3997fef9
...@@ -58,7 +58,6 @@ typedef struct TTAChannel { ...@@ -58,7 +58,6 @@ typedef struct TTAChannel {
typedef struct TTAContext { typedef struct TTAContext {
AVCodecContext *avctx; AVCodecContext *avctx;
AVFrame frame;
GetBitContext gb; GetBitContext gb;
const AVCRC *crc_table; const AVCRC *crc_table;
...@@ -303,15 +302,13 @@ static av_cold int tta_decode_init(AVCodecContext * avctx) ...@@ -303,15 +302,13 @@ static av_cold int tta_decode_init(AVCodecContext * avctx)
return -1; return -1;
} }
avcodec_get_frame_defaults(&s->frame);
avctx->coded_frame = &s->frame;
return 0; return 0;
} }
static int tta_decode_frame(AVCodecContext *avctx, void *data, static int tta_decode_frame(AVCodecContext *avctx, void *data,
int *got_frame_ptr, AVPacket *avpkt) int *got_frame_ptr, AVPacket *avpkt)
{ {
AVFrame *frame = data;
const uint8_t *buf = avpkt->data; const uint8_t *buf = avpkt->data;
int buf_size = avpkt->size; int buf_size = avpkt->size;
TTAContext *s = avctx->priv_data; TTAContext *s = avctx->priv_data;
...@@ -327,15 +324,15 @@ static int tta_decode_frame(AVCodecContext *avctx, void *data, ...@@ -327,15 +324,15 @@ static int tta_decode_frame(AVCodecContext *avctx, void *data,
init_get_bits(&s->gb, buf, buf_size*8); init_get_bits(&s->gb, buf, buf_size*8);
/* get output buffer */ /* get output buffer */
s->frame.nb_samples = framelen; frame->nb_samples = framelen;
if ((ret = ff_get_buffer(avctx, &s->frame)) < 0) { if ((ret = ff_get_buffer(avctx, frame)) < 0) {
av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n"); av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
return ret; return ret;
} }
// decode directly to output buffer for 24-bit sample format // decode directly to output buffer for 24-bit sample format
if (s->bps == 3) if (s->bps == 3)
s->decode_buffer = (int32_t *)s->frame.data[0]; s->decode_buffer = (int32_t *)frame->data[0];
// init per channel states // init per channel states
for (i = 0; i < s->channels; i++) { for (i = 0; i < s->channels; i++) {
...@@ -424,7 +421,7 @@ static int tta_decode_frame(AVCodecContext *avctx, void *data, ...@@ -424,7 +421,7 @@ static int tta_decode_frame(AVCodecContext *avctx, void *data,
i++; i++;
// check for last frame // check for last frame
if (i == s->last_frame_length && get_bits_left(&s->gb) / 8 == 4) { if (i == s->last_frame_length && get_bits_left(&s->gb) / 8 == 4) {
s->frame.nb_samples = framelen = s->last_frame_length; frame->nb_samples = framelen = s->last_frame_length;
break; break;
} }
} }
...@@ -439,12 +436,12 @@ static int tta_decode_frame(AVCodecContext *avctx, void *data, ...@@ -439,12 +436,12 @@ static int tta_decode_frame(AVCodecContext *avctx, void *data,
// convert to output buffer // convert to output buffer
if (s->bps == 2) { if (s->bps == 2) {
int16_t *samples = (int16_t *)s->frame.data[0]; int16_t *samples = (int16_t *)frame->data[0];
for (p = s->decode_buffer; p < s->decode_buffer + (framelen * s->channels); p++) for (p = s->decode_buffer; p < s->decode_buffer + (framelen * s->channels); p++)
*samples++ = *p; *samples++ = *p;
} else { } else {
// shift samples for 24-bit sample format // shift samples for 24-bit sample format
int32_t *samples = (int32_t *)s->frame.data[0]; int32_t *samples = (int32_t *)frame->data[0];
for (i = 0; i < framelen * s->channels; i++) for (i = 0; i < framelen * s->channels; i++)
*samples++ <<= 8; *samples++ <<= 8;
// reset decode buffer // reset decode buffer
...@@ -452,7 +449,6 @@ static int tta_decode_frame(AVCodecContext *avctx, void *data, ...@@ -452,7 +449,6 @@ static int tta_decode_frame(AVCodecContext *avctx, void *data,
} }
*got_frame_ptr = 1; *got_frame_ptr = 1;
*(AVFrame *)data = s->frame;
return buf_size; return buf_size;
error: error:
......
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