mirror of
https://git.ffmpeg.org/ffmpeg.git
synced 2025-03-04 03:28:05 +00:00
tta: decode directly to the user-provided AVFrame
This commit is contained in:
parent
3997fef952
commit
ad2104ba44
@ -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)
|
|||||||
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,
|
|||||||
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,
|
|||||||
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,20 +436,19 @@ 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
|
||||||
s->decode_buffer = NULL;
|
s->decode_buffer = NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
*got_frame_ptr = 1;
|
*got_frame_ptr = 1;
|
||||||
*(AVFrame *)data = s->frame;
|
|
||||||
|
|
||||||
return buf_size;
|
return buf_size;
|
||||||
error:
|
error:
|
||||||
|
Loading…
Reference in New Issue
Block a user