mirror of https://git.ffmpeg.org/ffmpeg.git
mmal: drop the h264 BSF
We can send mp4-style data directly. But for some reason, this requires sending the extradata as buffer with MMAL_BUFFER_HEADER_FLAG_CONFIG set. Reuse the infrastructure for sending AVPackets to do this. Signed-off-by: Luca Barbato <lu_zero@gentoo.org>
This commit is contained in:
parent
7bb1c1bfd2
commit
f290e48d86
|
@ -67,8 +67,6 @@ typedef struct MMALDecodeContext {
|
|||
AVClass *av_class;
|
||||
int extra_buffers;
|
||||
|
||||
AVBitStreamFilterContext *bsfc;
|
||||
|
||||
MMAL_COMPONENT_T *decoder;
|
||||
MMAL_QUEUE_T *queue_decoded_frames;
|
||||
MMAL_POOL_T *pool_in;
|
||||
|
@ -86,6 +84,7 @@ typedef struct MMALDecodeContext {
|
|||
int64_t frames_output;
|
||||
int eos_received;
|
||||
int eos_sent;
|
||||
int extradata_sent;
|
||||
} MMALDecodeContext;
|
||||
|
||||
// Assume decoder is guaranteed to produce output after at least this many
|
||||
|
@ -165,7 +164,7 @@ static void ffmmal_stop_decoder(AVCodecContext *avctx)
|
|||
}
|
||||
ctx->waiting_buffers_tail = NULL;
|
||||
|
||||
ctx->frames_output = ctx->eos_received = ctx->eos_sent = ctx->packets_sent = 0;
|
||||
ctx->frames_output = ctx->eos_received = ctx->eos_sent = ctx->packets_sent = ctx->extradata_sent = 0;
|
||||
}
|
||||
|
||||
static av_cold int ffmmal_close_decoder(AVCodecContext *avctx)
|
||||
|
@ -181,9 +180,6 @@ static av_cold int ffmmal_close_decoder(AVCodecContext *avctx)
|
|||
mmal_pool_destroy(ctx->pool_in);
|
||||
ffmmal_poolref_unref(ctx->pool_out);
|
||||
|
||||
if (ctx->bsfc)
|
||||
av_bitstream_filter_close(ctx->bsfc);
|
||||
|
||||
mmal_vc_deinit();
|
||||
|
||||
return 0;
|
||||
|
@ -356,18 +352,6 @@ static av_cold int ffmmal_init_decoder(AVCodecContext *avctx)
|
|||
format_in->es->video.par.den = avctx->sample_aspect_ratio.den;
|
||||
format_in->flags = MMAL_ES_FORMAT_FLAG_FRAMED;
|
||||
|
||||
if (avctx->codec->id == AV_CODEC_ID_H264 && avctx->extradata && avctx->extradata[0] == 1) {
|
||||
uint8_t *dummy_p;
|
||||
int dummy_int;
|
||||
ctx->bsfc = av_bitstream_filter_init("h264_mp4toannexb");
|
||||
if (!ctx->bsfc) {
|
||||
av_log(avctx, AV_LOG_ERROR, "Cannot open the h264_mp4toannexb BSF!\n");
|
||||
ret = AVERROR(ENOSYS);
|
||||
goto fail;
|
||||
}
|
||||
av_bitstream_filter_filter(ctx->bsfc, avctx, NULL, &dummy_p, &dummy_int, NULL, 0, 0);
|
||||
}
|
||||
|
||||
if (avctx->extradata_size) {
|
||||
if ((status = mmal_format_extradata_alloc(format_in, avctx->extradata_size)))
|
||||
goto fail;
|
||||
|
@ -442,7 +426,9 @@ fail:
|
|||
// (due to us not reading/returning enough output buffers) and won't accept
|
||||
// new input. (This wouldn't be an issue if MMAL input buffers always were
|
||||
// complete frames - then the input buffer just would have to be big enough.)
|
||||
static int ffmmal_add_packet(AVCodecContext *avctx, AVPacket *avpkt)
|
||||
// If is_extradata is set, send it as MMAL_BUFFER_HEADER_FLAG_CONFIG.
|
||||
static int ffmmal_add_packet(AVCodecContext *avctx, AVPacket *avpkt,
|
||||
int is_extradata)
|
||||
{
|
||||
MMALDecodeContext *ctx = avctx->priv_data;
|
||||
AVBufferRef *buf = NULL;
|
||||
|
@ -452,23 +438,12 @@ static int ffmmal_add_packet(AVCodecContext *avctx, AVPacket *avpkt)
|
|||
int ret = 0;
|
||||
|
||||
if (avpkt->size) {
|
||||
if (ctx->bsfc) {
|
||||
uint8_t *tmp_data;
|
||||
int tmp_size;
|
||||
if ((ret = av_bitstream_filter_filter(ctx->bsfc, avctx, NULL,
|
||||
&tmp_data, &tmp_size,
|
||||
avpkt->data, avpkt->size,
|
||||
avpkt->flags & AV_PKT_FLAG_KEY)) < 0)
|
||||
goto done;
|
||||
buf = av_buffer_create(tmp_data, tmp_size, NULL, NULL, 0);
|
||||
if (avpkt->buf) {
|
||||
buf = av_buffer_ref(avpkt->buf);
|
||||
} else {
|
||||
if (avpkt->buf) {
|
||||
buf = av_buffer_ref(avpkt->buf);
|
||||
} else {
|
||||
buf = av_buffer_alloc(avpkt->size);
|
||||
if (buf)
|
||||
memcpy(buf->data, avpkt->data, avpkt->size);
|
||||
}
|
||||
buf = av_buffer_alloc(avpkt->size);
|
||||
if (buf)
|
||||
memcpy(buf->data, avpkt->data, avpkt->size);
|
||||
}
|
||||
if (!buf) {
|
||||
ret = AVERROR(ENOMEM);
|
||||
|
@ -476,7 +451,8 @@ static int ffmmal_add_packet(AVCodecContext *avctx, AVPacket *avpkt)
|
|||
}
|
||||
size = buf->size;
|
||||
data = buf->data;
|
||||
ctx->packets_sent++;
|
||||
if (!is_extradata)
|
||||
ctx->packets_sent++;
|
||||
} else {
|
||||
if (!ctx->packets_sent) {
|
||||
// Short-cut the flush logic to avoid upsetting MMAL.
|
||||
|
@ -498,6 +474,9 @@ static int ffmmal_add_packet(AVCodecContext *avctx, AVPacket *avpkt)
|
|||
buffer->data = data;
|
||||
buffer->length = FFMIN(size, ctx->decoder->input[0]->buffer_size);
|
||||
|
||||
if (is_extradata)
|
||||
buffer->flags |= MMAL_BUFFER_HEADER_FLAG_CONFIG;
|
||||
|
||||
if (data == start)
|
||||
buffer->flags |= MMAL_BUFFER_HEADER_FLAG_FRAME_START;
|
||||
|
||||
|
@ -733,10 +712,21 @@ done:
|
|||
static int ffmmal_decode(AVCodecContext *avctx, void *data, int *got_frame,
|
||||
AVPacket *avpkt)
|
||||
{
|
||||
MMALDecodeContext *ctx = avctx->priv_data;
|
||||
AVFrame *frame = data;
|
||||
int ret = 0;
|
||||
|
||||
if ((ret = ffmmal_add_packet(avctx, avpkt)) < 0)
|
||||
if (avctx->extradata_size && !ctx->extradata_sent) {
|
||||
AVPacket pkt = {0};
|
||||
av_init_packet(&pkt);
|
||||
pkt.data = avctx->extradata;
|
||||
pkt.size = avctx->extradata_size;
|
||||
ctx->extradata_sent = 1;
|
||||
if ((ret = ffmmal_add_packet(avctx, &pkt, 1)) < 0)
|
||||
return ret;
|
||||
}
|
||||
|
||||
if ((ret = ffmmal_add_packet(avctx, avpkt, 0)) < 0)
|
||||
return ret;
|
||||
|
||||
if ((ret = ffmmal_fill_input_port(avctx)) < 0)
|
||||
|
|
Loading…
Reference in New Issue